package com.revai.engine;

import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.api.ProgressCallback;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.internal.DefaultShellCallback;

import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * MBG代码生成工具
 * Created by macro on 2018/4/26.
 */
public class Generator {
    public static void main(String[] args) throws Exception {
        // MBG 执行过程中的警告信息
        List<String> warnings = new ArrayList<String>();
        // 当生成的代码重复时，覆盖原代码
        boolean overwrite = true;
        // 读取我们的 MBG 配置文件
        InputStream is = Generator.class.getResourceAsStream("/generatorConfig.xml");
        ConfigurationParser cp = new ConfigurationParser(warnings);
        Configuration config = cp.parseConfiguration(is);
        is.close();

        DefaultShellCallback callback = new DefaultShellCallback(overwrite);
        ProgressCallback progressCallback = new MyProgressCallback();
        // 创建 MBG
        MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
        // 执行生成代码
        myBatisGenerator.generate(progressCallback);
        // 输出警告信息
        for (String warning : warnings) {
            System.out.printf(warning);
        }
    }

    static class MyProgressCallback implements ProgressCallback {
        @Override
        public void startTask(String taskName) {
            Matcher matcher = Pattern.compile("\\b\\w+\\.java\\b").matcher(taskName);
            if (matcher.find()) {
                String fileName = matcher.group().split("\\.")[0];
                if (!fileName.endsWith("Example") && !fileName.endsWith("Mapper")) {
                    // 执行生成service && controller
                    // CodeGenerator.genCode(fileName);
                }
            }
        }
    }
}
